Picture for Yifei Wang

Yifei Wang

Amazon AGI SF Lab

No More K-means:Single-Stage Sparse Coding for Efficient Multi-Vector Retrieval

Add code
May 28, 2026
Viaarxiv icon

Unbiased Diffusion Variational Inversion via Principled Posterior Matching

Add code
May 24, 2026
Viaarxiv icon

QUEST: Training Frontier Deep Research Agents with Fully Synthetic Tasks

Add code
May 22, 2026
Viaarxiv icon

Trusted Weights, Treacherous Optimizations? Optimization-Triggered Backdoor Attacks on LLMs

Add code
May 20, 2026
Viaarxiv icon

Data Difficulty and the Generalization--Extrapolation Tradeoff in LLM Fine-Tuning

Add code
May 13, 2026
Viaarxiv icon

Taming Outlier Tokens in Diffusion Transformers

Add code
May 06, 2026
Viaarxiv icon

Joint Energy Management and Coordinated AIGC Workload Scheduling for Distributed Data Centers: A Diffusion-Aided Reward Shaping Approach

Add code
May 03, 2026
Viaarxiv icon

Advancing Ligand-based Virtual Screening and Molecular Generation with Pretrained Molecular Embedding Distance

Add code
Apr 27, 2026
Viaarxiv icon

Reward Hacking in the Era of Large Models: Mechanisms, Emergent Misalignment, Challenges

Add code
Apr 15, 2026
Viaarxiv icon

EMMa: End-Effector Stability-Oriented Mobile Manipulation for Tracked Rescue Robots

Add code
Apr 09, 2026
Viaarxiv icon